numpy seaborn matplotlib tensorflow scikit-learn tensorflow-datasets==4.0 tqdm autograd